package cc.code.shapp.utils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;

public class ScreenCaptureUtil {

    private String fileName; //文件的前缀

    private String defaultName = "GuiCamera";

    static int serialNum = 0;

    private String imageFormat; //图像文件的格式

    private String defaultImageFormat="png";

    Dimension d = Toolkit.getDefaultToolkit().getScreenSize();

    /****************************************************************
     * 默认的文件前缀为GuiCamera，文件格式为PNG格式
     * The default construct will use the default
     * Image file surname "GuiCamera",
     * and default image format "png"
     ****************************************************************/
    public ScreenCaptureUtil() {
        fileName = defaultName;
        imageFormat = defaultImageFormat;
    }

    /****************************************************************
     * @param s the surname of the snapshot file
     * @param format the format of the  image file,
     * it can be "jpg" or "png"
     * 本构造支持JPG和PNG文件的存储
     ****************************************************************/
    public ScreenCaptureUtil(String s,String format) {
        fileName = s;
        imageFormat = format;
    }

    /****************************************************************
     * 对屏幕进行拍照
     * snapShot the Gui once
     ****************************************************************/
    public void snapShot(int x1, int y1, int x2, int y2)
    {
        try
        {
            //拷贝屏幕到一个BufferedImage对象screenshot
            BufferedImage screenshot = (new Robot()).createScreenCapture(new
                    Rectangle(x1, y1, x2-x1, y2-y1));
            serialNum++;
            //根据文件前缀变量和文件格式变量，自动生成文件名
            String name = fileName + String.valueOf(serialNum) + "." + imageFormat;
            File f = new File(name);
            System.out.print("Save File " + name);
            //将screenshot对象写入图像文件
            ImageIO.write(screenshot, imageFormat, f);
            System.out.print("..Finished!\n");
        }
        catch (Exception ex) {
            System.out.println(ex);
        }
    }

    public static void main(String[] a) throws Exception {
        Thread.sleep(3000);
        ScreenCaptureUtil ScreenCapture = new ScreenCaptureUtil("e:\\", "jpg");
        //ScreenCapture.snapShot();
    }
}
